
# 写一个程序，实现复制文件的功能
# 运行效果:
# 请输入要复制的源文件：/home/tarena/桌面/01_month01/day17_code/friends.txt
# 请输入复制到的目标文件: /home/tarena/桌面/a.txt
# 提示:
#    文件复制成功。

src_file = input('请输入要复制的源文件：')
dst_file = input('请输入复制到的目标文件：')

# fr = open(src_file, 'rb')
# fw = open(dst_file, 'wb')
try:
    with open(src_file, 'rb') as fr, open(dst_file, 'wb') as fw:
        # 循环读取数据，每次读取1M数据，直到没有数据可以读为止
        while True:
            b = fr.read(1 * 1024*1024)
            if not b:  # 如果b绑定空字节串,退出复制
                break
            fw.write(b)
        # fw.write(fr.read())
except OSError:
    print('复制失败！')
else:
    print('复制成功')
# fw.close()
# fr.close()